package ds_industry_2025.ds.YangJuan_2024.T1

import org.apache.spark.sql.SparkSession
import org.apache.spark.sql.functions.{col, greatest, lit, max}

import java.text.SimpleDateFormat
import java.util.{Calendar, Properties}

/*
  抽取 shtd_store 库中 order_info 的增量数据进入 Hive 的 ods 库中表 order_info，根据
ods.order_info 表中 operate_time 或 create_time 作为增量字段(即 MySQL 中每条数据取
这两个时间中较大的那个时间作为增量字段去和 ods 里的这两个字段中较大的时间进
行比较)，只将新增的数据抽入，字段名称、类型不变，同时添加静态分区，分区字段
为 etl_date，类型为 String，且值为当前比赛日的前一天日期（分区字段格式为
yyyyMMdd）。使用 hive cli 执行 show partitions ods.order_info 命令，将结果截图粘贴
至客户端桌面【Release\任务 B 提交结果.docx】中对应的任务序号下；
 */
object t5 {
  def main(args: Array[String]): Unit = {
    val spark = SparkSession.builder()
      .master("local[*]")
      .appName("T5")
      .config("hive.exec.dynamic.partition.mode","nonstrict")
      .config("spark.serializer","org.apache.spark.serializer.KryoSerializer")
      .config("spark.sql.extensions","org.apache.spark.sql.hudi.HoodieSparkSessionExtension")
      .enableHiveSupport()
      .getOrCreate()

    val conn = new Properties()
    conn.setProperty("user", "root")
    conn.setProperty("password", "123456")
    conn.setProperty("driver", "com.mysql.jdbc.Driver")

    val day = Calendar.getInstance()
    day.add(Calendar.DATE, -1)
    val yesterday = new SimpleDateFormat("yyyyMMdd").format(day.getTime)

    val max_time = spark.table("ods.order_info")
      .agg(max(greatest(col("create_time"), col("operate_time"))))
      .collect()(0).get(0).toString

    spark.read.jdbc("jdbc:mysql://192.168.40.110:3306/shtd_store?useSSL=false", "order_info", conn)
      .where(
        greatest(col("create_time"), col("operate_time")) > lit(max_time).cast("timestamp")
      )
      .withColumn("etl_date", lit(yesterday))
      .write.mode("append")
      .partitionBy("etl_date")
      .saveAsTable("ods.order_info")

    spark.close()
  }

}
